#导包
from  selenium import webdriver
import  time
#创建一个对象
from selenium.webdriver.common.by import By


class TestCase:
    #初始化drive方法
     def __init__(self):
         self.driver=webdriver.Chrome()

# 请求so页面
     def git_so(self):
         self.driver.get('https://www.so.com/')
         # 定位搜索框,写入文本信息
         self.driver.find_element(By.ID,'input').send_keys('python自动化测试')

     def diwei(self):
         # 定位到搜索按钮,点击
         self.driver.find_element(By.ID,'search-button').click()
          # 获取页面标题
         print(self.driver.title)
         # 判断python是否在页面标题中
         if 'python' in  self.driver.title:
             print('找到了')
         else:
             print('没有找到')


     def quit(self):
        self.driver.quit()



if __name__ == '__main__':
    add =TestCase()
    add.git_so()
    add.diwei()
    time.sleep(3)
    add.quit()


